package cn.wolfcode.luowowo.webSite.controller;

import cn.wolfcode.luowowo.article.query.TravelCommendQuery;
import cn.wolfcode.luowowo.article.query.TravelQuery;
import cn.wolfcode.luowowo.article.service.IStrategyCommendService;
import cn.wolfcode.luowowo.article.service.ITravelCommendService;
import cn.wolfcode.luowowo.article.service.ITravelService;
import cn.wolfcode.luowowo.cache.service.IUserInfoRedisService;
import cn.wolfcode.luowowo.member.domain.UserInfo;
import cn.wolfcode.luowowo.search.query.SearchQueryObject;
import cn.wolfcode.luowowo.search.service.*;
import cn.wolfcode.luowowo.search.template.DestinationTemplate;
import cn.wolfcode.luowowo.search.template.StrategyTemplate;
import cn.wolfcode.luowowo.search.template.TravelTemplate;
import cn.wolfcode.luowowo.search.template.UserInfoTemplate;
import cn.wolfcode.luowowo.search.vo.SearchResult;
import cn.wolfcode.luowowo.webSite.annotation.RequireLogin;
import cn.wolfcode.luowowo.webSite.annotation.UserParam;
import cn.wolfcode.luowowo.webSite.util.CookieUtil;
import com.alibaba.dubbo.config.annotation.Reference;
import com.github.pagehelper.PageInfo;
import org.elasticsearch.search.SearchService;
import org.springframework.data.domain.Page;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.servlet.http.HttpServletRequest;
import java.util.List;

@Controller
public class IndexController {

    @Reference
    private IUserInfoRedisService userInfoRedisService;

    @Reference
    private ITravelCommendService travelCommendService;

    @Reference
    private IStrategyCommendService strategyCommendService;

    @Reference
    private ITravelService travelService;

    @Reference
    private IDestinationTemplateSearchService destinationTemplateSearchService;

    @Reference
    private IStrategyTemplateSearchService strategyTemplateSearchService;

    @Reference
    private ITravelTemplateSearchService travelTemplateSearchService;

    @Reference
    private IUserInfoTemplateSearchService userInfoTemplateSearchService;

    @Reference
    private ISearchService searchService;

    @RequestMapping("")
//    @RequireLogin
    public String index(HttpServletRequest request, Model model){
        String token = CookieUtil.getToken(request);
        //从Redis里面取出当前用户
        UserInfo userInfo = userInfoRedisService.getUserByToken(token);
        //共享当前登录用户信息
        request.getSession().setAttribute("userInfo",userInfo);

        //首页游记推荐显示
        model.addAttribute("tcs", travelCommendService.getTop5());

        //首页攻略推荐显示
        model.addAttribute("sc", strategyCommendService.getTop5().get(0));

        return "/index/index";
    }

    @RequestMapping("index/travelPage")
    public String travelPage(TravelQuery qo,Model model){
        PageInfo pageInfo = travelService.query(qo);
        model.addAttribute("pageInfo", pageInfo);
        return "index/travelPageTpl";
    }

    @RequestMapping("/q")
    public String search(@ModelAttribute("qo") SearchQueryObject qo, Model model){

        switch (qo.getType()){
            case SearchQueryObject.SEARCH_TYPE_DEST:
                return destSearch(model, qo);
            case SearchQueryObject.SEARCH_TYPE_STRATEGY:
                return strategySearch(model, qo);
            case SearchQueryObject.SEARCH_TYPE_TRAVEL:
                return travelSearch(model, qo);
            case SearchQueryObject.SEARCH_TYPE_USER:
                return userSearch(model, qo);
            default:
                return allSearch(model, qo);
        }
    }

    //全部搜索
    private String allSearch(Model model, SearchQueryObject qo) {
        Page<TravelTemplate> travels = searchService.searchWithHighlight(TravelTemplate.INDEX_NAME, TravelTemplate.TYPE_NAME,
                TravelTemplate.class, qo, "title", "summary");

        Page<StrategyTemplate> strategys = searchService.searchWithHighlight(StrategyTemplate.INDEX_NAME, StrategyTemplate.TYPE_NAME,
                StrategyTemplate.class, qo, "title", "subTitle","summary");

        Page<UserInfoTemplate> users = searchService.searchWithHighlight(UserInfoTemplate.INDEX_NAME, UserInfoTemplate.TYPE_NAME,
                UserInfoTemplate.class, qo, "nickName", "destName");

        Page<DestinationTemplate> dests = searchService.searchWithHighlight(DestinationTemplate.INDEX_NAME, DestinationTemplate.TYPE_NAME,
                DestinationTemplate.class, qo, "name", "info");

        SearchResult data = new SearchResult();

        data.setUsers(users.getContent());
        data.setTravels(travels.getContent());
        data.setStrategys(strategys.getContent());
        data.setDests(dests.getContent());

        data.setTotal(dests.getTotalElements()+users.getTotalElements()+travels.getTotalElements()+strategys.getTotalElements());

        model.addAttribute("data", data);
        return "/index/searchAll";
    }

    //游记搜索
    private String travelSearch(Model model, SearchQueryObject qo) {
        Page<TravelTemplate> page = searchService.searchWithHighlight(TravelTemplate.INDEX_NAME, TravelTemplate.TYPE_NAME,
                TravelTemplate.class, qo, "title", "summary");
        model.addAttribute("page", page);
        return "/index/searchTravel";
    }

    //攻略搜索
    private String strategySearch(Model model, SearchQueryObject qo) {
        Page<StrategyTemplate> page = searchService.searchWithHighlight(StrategyTemplate.INDEX_NAME, StrategyTemplate.TYPE_NAME,
                StrategyTemplate.class, qo, "title", "subTitle","summary");
        model.addAttribute("page", page);
        return "/index/searchStrategy";
    }

    //用户搜索
    private String userSearch(Model model, SearchQueryObject qo) {
        Page<UserInfoTemplate> page = searchService.searchWithHighlight(UserInfoTemplate.INDEX_NAME, UserInfoTemplate.TYPE_NAME,
                UserInfoTemplate.class, qo, "nickName", "destName");
        model.addAttribute("page", page);
        return "/index/searchUser";
    }

    private String destSearch(Model model, SearchQueryObject qo) {

        SearchResult data = new SearchResult();

        //查询目的地对象
        DestinationTemplate dest = destinationTemplateSearchService.findByName(qo.getKeyword());
        if(dest!=null){
            long total = 0;
            //查询目的地下所有攻略,前5个
            List<StrategyTemplate> sts = strategyTemplateSearchService.findByDestName(dest.getName());
            total+=sts.size();
            if(sts.size()>5){
                sts=sts.subList(0, 5);
            }

            //查询目的地下所有游记,前5个
            List<TravelTemplate> ts = travelTemplateSearchService.findByDestName(dest.getName());
            total+=ts.size();
            if(ts.size()>5){
                ts=ts.subList(0, 5);
            }

            //查询目的地下的所有用户,前5个
            List<UserInfoTemplate> us = userInfoTemplateSearchService.findByDestName(dest.getName());
            total+=us.size();
            if(us.size()>5){
                us=us.subList(0, 5);
            }
            data.setStrategys(sts);
            data.setTravels(ts);
            data.setUsers(us);
            data.setTotal(total);
        }
        model.addAttribute("data", data);
        model.addAttribute("dest", dest);

        return "/index/searchDest";
    }
}
